package com.hethongviet.job.helpers.sound;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

import com.hethongviet.job.AppMain;

import android.annotation.SuppressLint;
import android.media.AudioFormat;
import android.media.AudioRecord;
import android.media.MediaRecorder;
import android.os.Environment;

@SuppressLint("NewApi")
public class RecorderWave
{

	private static final int RECORDER_SAMPLERATE = 8000;

	public static final long MAX_RECORD_TIME = (long) ((RECORDER_SAMPLERATE / 8000 * 1.0) * 60000);// miliseconds

	private static final int RECORDER_BPP = 16;
	private static final String AUDIO_RECORDER_FILE_EXT_WAV = ".wav";
	private static final String AUDIO_RECORDER_TEMP_FILE = "record_temp.raw";

	private static final int RECORDER_CHANNELS = AudioFormat.CHANNEL_IN_STEREO;
	private static final int RECORDER_AUDIO_ENCODING = AudioFormat.ENCODING_PCM_16BIT;

	private AudioRecord recorder = null;
	private int bufferSize = 0;
	private Thread recordingThread = null;
	private boolean isRecording = false;

	public RecorderWave()
	{
	}

	public String getName()
	{
		return ("quanlytiendo" + AUDIO_RECORDER_FILE_EXT_WAV);
	}

	public String getFolder()
	{
		return new File(Environment.getExternalStorageDirectory().getPath()+ AppMain.DIRECTORY_AUDIO_FILE).getPath();
	}

	public String getFileName()
	{
		return (createFolderAudio() + "/" + getName());
	}

	private String createFolderAudio()
	{
		File folder = new File(getFolder());

		if (!folder.exists())
		{
			folder.mkdirs();
		}
		return folder.getPath();
	}

	private String getTempFilename()
	{
		File file = new File(createFolderAudio(), AUDIO_RECORDER_TEMP_FILE);
		return (file.getAbsolutePath());
	}

	public void startRecording()
	{
		bufferSize = AudioRecord.getMinBufferSize(RECORDER_SAMPLERATE,
				RECORDER_CHANNELS, RECORDER_AUDIO_ENCODING);

		recorder = new AudioRecord(MediaRecorder.AudioSource.MIC,
				RECORDER_SAMPLERATE, RECORDER_CHANNELS,
				RECORDER_AUDIO_ENCODING, bufferSize);

		recorder.startRecording();

		isRecording = true;

		recordingThread = new Thread(new Runnable()
		{

			@Override
			public void run()
			{
				writeAudioDataToFile();
			}
		}, "AudioRecorder Thread");

		recordingThread.start();
	}

	public boolean isRecording()
	{
		if (recorder != null
				&& recorder.getRecordingState() == AudioRecord.RECORDSTATE_RECORDING)
			return true;
		return false;
	}

	private void writeAudioDataToFile()
	{
		byte data[] = new byte[bufferSize];
		String filename = getTempFilename();
		FileOutputStream os = null;

		try
		{
			os = new FileOutputStream(filename);
		} catch (FileNotFoundException e)
		{
			e.printStackTrace();
		}

		int read = 0;

		if (null != os)
		{
			while (isRecording)
			{
				read = recorder.read(data, 0, bufferSize);

				if (AudioRecord.ERROR_INVALID_OPERATION != read)
				{
					try
					{
						os.write(data);
					} catch (IOException e)
					{
						e.printStackTrace();
					}
				}
			}

			try
			{
				os.close();
			} catch (IOException e)
			{
				e.printStackTrace();
			}
		}
	}

	public void stopRecording()
	{
		if (null != recorder)
		{
			isRecording = false;

			recorder.stop();
			recorder.release();

			recorder = null;
			recordingThread = null;
		}

		if (isExistTempFile())
		{
			AppMain.debug("copyWaveFile( " + getTempFilename() + ","
					+ getFileName() + ")");

			copyWaveFile(getTempFilename(), getFileName());
			deleteTempFile();
		}
	}

	private void deleteTempFile()
	{
		File file = new File(getTempFilename());

		file.delete();
	}

	private boolean isExistTempFile()
	{
		File file = new File(getTempFilename());
		return file.exists();
	}

	public void copyWaveFile(String inFilename, String outFilename)
	{
		FileInputStream in = null;
		FileOutputStream out = null;
		long totalAudioLen = 0;
		long totalDataLen = totalAudioLen + 36;
		long longSampleRate = RECORDER_SAMPLERATE;
		int channels = 2;
		long byteRate = RECORDER_BPP * RECORDER_SAMPLERATE * channels / 8;

		byte[] data = new byte[bufferSize];

		try
		{
			in = new FileInputStream(inFilename);
			out = new FileOutputStream(outFilename);
			totalAudioLen = in.getChannel().size();
			totalDataLen = totalAudioLen + 36;

			WriteWaveFileHeader(out, totalAudioLen, totalDataLen,
					longSampleRate, channels, byteRate);

			while (in.read(data) != -1)
			{
				out.write(data);
			}

			in.close();
			out.close();
		} catch (FileNotFoundException e)
		{
			e.printStackTrace();
		} catch (IOException e)
		{
			e.printStackTrace();
		}
	}

	private void WriteWaveFileHeader(FileOutputStream out, long totalAudioLen,
			long totalDataLen, long longSampleRate, int channels, long byteRate)
			throws IOException
	{

		byte[] header = new byte[44];

		header[0] = 'R'; // RIFF/WAVE header
		header[1] = 'I';
		header[2] = 'F';
		header[3] = 'F';
		header[4] = (byte) (totalDataLen & 0xff);
		header[5] = (byte) ((totalDataLen >> 8) & 0xff);
		header[6] = (byte) ((totalDataLen >> 16) & 0xff);
		header[7] = (byte) ((totalDataLen >> 24) & 0xff);
		header[8] = 'W';
		header[9] = 'A';
		header[10] = 'V';
		header[11] = 'E';
		header[12] = 'f'; // 'fmt ' chunk
		header[13] = 'm';
		header[14] = 't';
		header[15] = ' ';
		header[16] = 16; // 4 bytes: size of 'fmt ' chunk
		header[17] = 0;
		header[18] = 0;
		header[19] = 0;
		header[20] = 1; // format = 1
		header[21] = 0;
		header[22] = (byte) channels;
		header[23] = 0;
		header[24] = (byte) (longSampleRate & 0xff);
		header[25] = (byte) ((longSampleRate >> 8) & 0xff);
		header[26] = (byte) ((longSampleRate >> 16) & 0xff);
		header[27] = (byte) ((longSampleRate >> 24) & 0xff);
		header[28] = (byte) (byteRate & 0xff);
		header[29] = (byte) ((byteRate >> 8) & 0xff);
		header[30] = (byte) ((byteRate >> 16) & 0xff);
		header[31] = (byte) ((byteRate >> 24) & 0xff);
		header[32] = (byte) (2 * 16 / 8); // block align
		header[33] = 0;
		header[34] = RECORDER_BPP; // bits per sample
		header[35] = 0;
		header[36] = 'd';
		header[37] = 'a';
		header[38] = 't';
		header[39] = 'a';
		header[40] = (byte) (totalAudioLen & 0xff);
		header[41] = (byte) ((totalAudioLen >> 8) & 0xff);
		header[42] = (byte) ((totalAudioLen >> 16) & 0xff);
		header[43] = (byte) ((totalAudioLen >> 24) & 0xff);

		out.write(header, 0, 44);
	}

}
